python avoid keyerror in dict